package Leetcode.搜索与回溯;

import Leetcode.树.TreeNode;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * @Author: kirito
 * @Date: 2024/8/26 16:37
 * @Description:
 * 给你二叉树的根节点 root 和一个整数目标和 targetSum ，
 * 找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
 *
 * 叶子节点 是指没有子节点的节点。
 *
 * 示例 1：
 *
 * 输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
 * 输出：[[5,4,11,2],[5,8,4,5]]
 */

public class 路径总和2 {
    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Deque<Integer> path = new LinkedList<Integer>();
        dfs(root, res, path, targetSum);
        return res;
    }

    private void dfs(TreeNode root, List<List<Integer>> res, Deque<Integer> path, int targetSum) {
        if (root == null) {
            return;
        }
        path.offerLast(root.val);
        targetSum-= root.val;
        if (root.left == null && root.right == null && targetSum == 0) {
            res.add(new LinkedList<>(path));
        }
        dfs(root.left, res, path, targetSum);
        dfs(root.right, res, path, targetSum);
        path.pollLast();
    }
}
